<template>
  <div>
    <vxe-button status="success" @click="getSelectEvent">获取已选</vxe-button>
    <vxe-gantt ref="ganttRef" v-bind="ganttOptions" v-on="ganttEvents"></vxe-gantt>
  </div>
</template>

<script lang="ts" setup>
import { ref, reactive } from 'vue'
import { VxeUI, VxeGanttInstance, VxeGanttProps, VxeGanttListeners } from 'vxe-gantt'

interface RowVO {
  id: number
  title: string
  start: string
  end: string
  progress: number
}

const ganttRef = ref<VxeGanttInstance<RowVO>>()

const allList = [
  { id: 10001, title: 'A项目', start: '2024-03-01', end: '2024-03-04', progress: 3 },
  { id: 10002, title: '城市道路修理进度', start: '2024-03-03', end: '2024-03-08', progress: 10 },
  { id: 10003, title: 'B大工程', start: '2024-03-03', end: '2024-03-11', progress: 90 },
  { id: 10004, title: '超级大工程', start: '2024-03-05', end: '2024-03-11', progress: 15 },
  { id: 10005, title: '地球净化项目', start: '2024-03-08', end: '2024-03-15', progress: 100 },
  { id: 10006, title: '一个小目标项目', start: '2024-03-10', end: '2024-03-21', progress: 0 },
  { id: 10007, title: '某某计划', start: '2024-03-15', end: '2024-03-24', progress: 70 },
  { id: 10008, title: '某某科技项目', start: '2024-03-20', end: '2024-03-29', progress: 50 },
  { id: 10009, title: '地铁建设工程', start: '2024-03-19', end: '2024-03-20', progress: 5 },
  { id: 10010, title: '公寓装修计划2', start: '2024-03-12', end: '2024-03-20', progress: 30 },
  { id: 10011, title: '两个小目标工程', start: '2024-03-01', end: '2024-03-04', progress: 20 },
  { id: 10012, title: '蓝天计划', start: '2024-03-02', end: '2024-03-08', progress: 50 },
  { id: 10013, title: 'C大项目', start: '2024-03-08', end: '2024-03-11', progress: 10 },
  { id: 10014, title: 'H计划', start: '2024-03-12', end: '2024-03-16', progress: 100 },
  { id: 10015, title: '铁路修建计划', start: '2024-03-05', end: '2024-03-06', progress: 0 },
  { id: 10016, title: 'D项目', start: '2024-03-06', end: '2024-03-11', progress: 10 },
  { id: 10017, title: '海外改造工程', start: '2024-03-08', end: '2024-03-09', progress: 0 },
  { id: 10018, title: 'Z计划', start: '2024-03-24', end: '2024-03-26', progress: 80 },
  { id: 10019, title: 'F工程', start: '2024-03-20', end: '2024-03-28', progress: 10 },
  { id: 10020, title: '投资大项目', start: '2024-03-23', end: '2024-03-28', progress: 60 },
  { id: 10021, title: 'X计划', start: '2024-03-16', end: '2024-03-25', progress: 10 },
  { id: 10022, title: '上天计划', start: '2024-03-05', end: '2024-03-24', progress: 0 },
  { id: 10023, title: 'G项目', start: '2024-03-08', end: '2024-03-28', progress: 5 },
  { id: 10024, title: '下地计划', start: '2024-03-09', end: '2024-03-16', progress: 50 }
]

const pagerVO = reactive({
  total: 0,
  currentPage: 1,
  pageSize: 10
})

const ganttOptions = reactive<VxeGanttProps<RowVO>>({
  showOverflow: true,
  border: true,
  loading: false,
  height: 500,
  rowConfig: {
    keyField: 'id'
  },
  checkboxConfig: {
    reserve: true,
    showReserveStatus: true
  },
  pagerConfig: pagerVO,
  taskBarConfig: {
    showProgress: true,
    showContent: true
  },
  taskViewConfig: {
    tableStyle: {
      width: 480
    }
  },
  columns: [
    { type: 'checkbox', width: 70 },
    { field: 'title', title: '任务名称' },
    { field: 'start', title: '开始时间', width: 100 },
    { field: 'end', title: '结束时间', width: 100 }
  ],
  data: []
})

// 模拟前端分页
const handlePageData = () => {
  ganttOptions.loading = true
  setTimeout(() => {
    const { pageSize, currentPage } = pagerVO
    pagerVO.total = allList.length
    ganttOptions.data = allList.slice((currentPage - 1) * pageSize, currentPage * pageSize)
    ganttOptions.loading = false
  }, 100)
}

const ganttEvents: VxeGanttListeners = {
  pageChange ({ pageSize, currentPage }) {
    pagerVO.currentPage = currentPage
    pagerVO.pageSize = pageSize
    handlePageData()
  }
}

const getSelectEvent = () => {
  const $gantt = ganttRef.value
  if ($gantt) {
    const selectRecords = $gantt.getCheckboxRecords()
    const selectReserveRecords = $gantt.getCheckboxReserveRecords()
    VxeUI.modal.message({
      content: `总共勾选： ${selectRecords.length + selectReserveRecords.length} 条，当前页勾选：${selectRecords.length} 条，已保留勾选：${selectReserveRecords.length} 条`,
      status: 'success'
    })
  }
}

handlePageData()
</script>
